package com.example.mq.mqserver.core;

import com.example.mq.common.Consumer;
import com.example.mq.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 这个类表示存储消息的队列
 */
@Data
public class MSGQueue {
    // 表示队列身份标识（唯一）
    private String name;
    // 表示队列是否持久化, true持久化,false 不持久化
    private boolean durable = false;


    // 这个属性为 true ,表示这个队列只能给一个消费者使用（别人用不了），如果为 false 则大家都可以用
    // 这个功能为未实现
    private boolean exclusive = false;
    // 为 true 表示没有使用，自动删除，false 则不会自动删除
    // 这个功能未实现
    private boolean autoDelete = false;
    // 也表示扩展参数，后续代码没有实现
    private Map<String, Object> arguments = new HashMap<>();

    // 当前队列都有哪些消费者订阅了
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();

    // 记录当前第几个消费者，方便实现轮询策略
    private AtomicInteger consumerSeq = new AtomicInteger(0);

    // 添加新的订阅者
    public void addConsumerEnv(ConsumerEnv consumerEnv) {
            consumerEnvList.add(consumerEnv);
    }

    // 删除订阅者

    // 挑选一个订阅者用来处理当前的消息，按照轮询
    public ConsumerEnv chooseConsumer() {
        if (consumerEnvList.size() == 0) {
            // 该队列没人订阅
            return null;
        }
        // 计算一下当前要取的元素下标
        int index = consumerSeq.get() % consumerEnvList.size();
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);
    }

    public String getArguments() {
        // 把当前 arguments 参数转换成 String ,基于 Json
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return "{}";
    }

    // 这个方法是从数据库中，读数据后，构造 Exchange 对象，会自动调用到
    public void setArguments(String argumentsJSon) {
        // 把参数中的 argumentsJson 按照 JSON 格式解析
        // 转换成上述 map 对象
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            this.arguments = objectMapper.readValue(argumentsJSon, new TypeReference<HashMap<String, Object>>() {
            });
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

    // 在这里针对 arguments, 再提供一组 getter setter, 用来更方便的获取/设置这里的键值对
    public Object getArguments(String key) {
        return arguments.get(key);
    }

    public void setArguments(String key, Object value) {
        arguments.put(key, value);
    }

    public void setArguments(Map<String, Object> arguments) {
        this.arguments = arguments;
    }
}
